43ca78fea0115e05c8d14626f470010e2f7334c3,tez-runtime-library/src/test/java/org/apache/tez/dag/library/vertexmanager/TestFairShuffleVertexManager.java,TestFairShuffleVertexManager,testFairSchedulingWithPartitionStats,#,146

Before Change


    Assert.assertEquals(2, edgeManager.getNumDestinationTaskPhysicalInputs(2));
    for (int sourceTaskIndex = 1; sourceTaskIndex < 3; sourceTaskIndex++) {
      for (int j = 0; j < 2; j++) {
        routeMetadata = (j == 0) ?
            edgeManager.routeCompositeDataMovementEventToDestination(
                sourceTaskIndex, 2) :
            edgeManager.routeInputSourceTaskFailedEventToDestination(
                sourceTaskIndex, 2);
        Assert.assertEquals(1, routeMetadata.getNumEvents());
        if (j == 0) {
          Assert.assertEquals(2, routeMetadata.getSourceIndices()[0]);
        }
        Assert.assertEquals(sourceTaskIndex - 1,
            routeMetadata.getTargetIndices()[0]);

After Change


    Assert.assertEquals(2, edgeManager.getNumDestinationTaskPhysicalInputs(2));
    for (int sourceTaskIndex = 1; sourceTaskIndex < 3; sourceTaskIndex++) {
      for (int j = 0; j < 2; j++) {
        if (j == 0) {
          EdgeManagerPluginOnDemand.CompositeEventRouteMetadata routeMetadata =
              edgeManager.routeCompositeDataMovementEventToDestination(sourceTaskIndex, 2);
          Assert.assertEquals(1, routeMetadata.getCount());
          Assert.assertEquals(2, routeMetadata.getSource());
          Assert.assertEquals(sourceTaskIndex - 1, routeMetadata.getTarget());
        } else {
          EdgeManagerPluginOnDemand.EventRouteMetadata routeMetadata =
              edgeManager.routeInputSourceTaskFailedEventToDestination(sourceTaskIndex, 2);
          Assert.assertEquals(1, routeMetadata.getNumEvents());
          Assert.assertEquals(sourceTaskIndex - 1, routeMetadata.getTargetIndices()[0]);
        }